package leetcode;
/**
 * 236.二叉树的最近公共祖先
 * 给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
 * 最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。
 *
 * 输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
 * 输出：3
 * 解释：节点 5 和节点 1 的最近公共祖先是节点 3
 */
public class Num_236 {
    //最近公共祖先
    private TreeNode lca;
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //找到p和q的所有祖先
        findNode(root,p,q);
        return lca;
    }
    //检验该二叉树中是否有p或q
    private boolean findNode(TreeNode root, TreeNode p, TreeNode q) {
        //边界条件
        if(root == null){
            return false;
        }
        //先检查根节点
        int cur = (root.val == p.val || root.val == q.val) ? 1 : 0;
        //再检验左子树
        int left = findNode(root.left,p,q) ? 1 : 0;
        //再检验右子树
        int right = findNode(root.right,p,q) ? 1 : 0;
        //当两个节点出现在根节点的左右两侧时，该根节点就是最近祖先
        if(cur + left + right == 2){
            lca = root;
        }
        //该根节点的二叉树出现p或q时，返回true
        return (cur + left + right) > 0;
    }
}
